import sys
import os

# 获取当前文件所在目录，将其加入系统路径，确保能正确导入 demo 下的模块
current_dir = os.path.dirname(os.path.abspath(__file__))
sys.path.append(current_dir)

def main():
    while True:
        print("\n===== 游戏选择平台 =====")
        print("1. 狼人杀")
        print("2. 谁是卧底")
        print("3. 退出程序")
        
        choice = input("请选择要启动的游戏 (1-3): ")
        
        if choice == '1':
            print("\n正在启动狼人杀游戏...")
            try:
                # 从 demo 目录下导入狼人杀游戏模块及对应的类
                from langrensha import WerewolfGame  
                # 创建狼人杀游戏实例并启动
                game = WerewolfGame()
                game.start()
                
            except ImportError as e:
                print(f"导入狼人杀游戏失败: 找不到相关模块 - {e}")
            except Exception as e:
                print(f"启动狼人杀游戏时发生错误: {e}")
        
        elif choice == '2':
            print("\n正在启动谁是卧底游戏...")
            try:
                # 从 demo 目录下导入谁是卧底游戏模块
                from sheishiwodi import main as sheishiwodi_main  
                # 调用谁是卧底游戏的 main 函数启动游戏
                sheishiwodi_main()
                
            except ImportError as e:
                print(f"导入谁是卧底游戏失败: 找不到相关模块 - {e}")
            except Exception as e:
                print(f"启动谁是卧底游戏时发生错误: {e}")
        
        elif choice == '3':
            print("\n感谢使用，再见！")
            break
        
        else:
            print("无效的选择，请输入 1-3 之间的数字")

if __name__ == "__main__":
    main()